package com.wn.generator;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

/**
 * 代码生成器
 */
public class ProductServiceGenerator {
    public static void main(String[] args) {

        FastAutoGenerator.create(
                "jdbc:mysql://localhost:3306/pms_product",
                "root",
                "root")

                .globalConfig(builder -> {
                    builder.author("yellow docker") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .dateType(DateType.ONLY_DATE)  //把SQL中的时间  只使用 java.util.date 代替
                            .disableOpenDir()
                            .outputDir("C:\\蜗牛\\深圳java6期\\maven-pro\\sk-sys\\sk-product-pservice\\sk-product-core\\src\\main\\java"); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.wn.sk.product.core") // 设置父包名
                            .moduleName(null) // 设置父包模块名
                            .pathInfo(Collections.singletonMap(
                                    OutputFile.mapperXml,
                                    "C:\\蜗牛\\深圳java6期\\maven-pro\\sk-sys\\sk-product-pservice\\sk-product-core\\src\\main\\resources\\mapper")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.entityBuilder().enableLombok();
                    builder.controllerBuilder()
                            .enableHyphenStyle()
                            .enableRestStyle();
                    // 把那些表进行代码生成
                    builder.addInclude(
                            "pms_product_cover", "pms_product","pms_promotion_seckill",
                            "pms_comment","pms_product_detail","pms_product_param") // 设置需要生成的表名
                            .addTablePrefix("pms"); // 设置过滤表前缀

                })
//                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();
    }
}